package model;

import network.Device;
import org.jgrapht.graph.DefaultEdge;

import java.util.HashSet;

public class LabeledEdge extends DefaultEdge {
    private Device from;
    private Device to;
    private HashSet<Integer> predicates;

    public LabeledEdge(Device from, Device to) {
        this.from = from;
        this.to = to;
        this.predicates = new HashSet<>();
    }

    public void addPredicate(int predicate) {
        this.predicates.add(predicate);
    }

    public HashSet<Integer> getPredicates() {
        return predicates;
    }

    public Device getFrom() {
        return from;
    }

    public Device getTo() {
        return to;
    }
}
